import unittest
import breadth_first_search as bf

class MyTestCase(unittest.TestCase):
    def test_something(self):
        graph = bf.Graph()
        graph.add_vertex("anuj").add_vertex("bob").add_vertex("you").add_vertex("peggy").add_vertex("alice").add_vertex(
            "claire").add_vertex("jonny").add_vertex("thom")
        (graph.add_edge("anuj", "bob").add_edge("bob", "you").add_edge("bob", "peggy").add_edge("peggy",
                                                                                                "alice").add_edge("you",
                                                                                                                  "alice").add_edge(
            "you", "claire")
         .add_edge("claire", "jonny").add_edge("claire", "thom"))
        # graph.bfs("anuj")
        print(graph.bfs_route("anuj", "thom"))
        # self.assertEqual(True, False)  # add assertion here


if __name__ == '__main__':
    unittest.main()
